docker: Add torchelastic to docker image (#45438) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/45438 Adds torchelastic (as well as its dependencies) to the official docker images Signed-off-by: Eli Uriegas <eliuriegas@fb.com> Test Plan: Imported from OSS Reviewed By: tierex Differential Revision: D23963787 Pulled By: seemethere fbshipit-source-id: 54ebb4b9c50699e543f264975dadf99badf55753
diff --git a/Dockerfile b/Dockerfile index 5bae3ec..d328a75 100644 --- a/Dockerfile +++ b/Dockerfile
@@ -52,6 +52,7 @@ ARG INSTALL_CHANNEL=pytorch-nightly RUN /opt/conda/bin/conda install -c "${INSTALL_CHANNEL}" -y pytorch torchvision cudatoolkit=11.0.221 && \ /opt/conda/bin/conda clean -ya +RUN /opt/conda/bin/pip install torchelastic FROM ${BASE_IMAGE} as official LABEL com.nvidia.volumes.needed="nvidia_driver" 